package com.eonmind.common.contract.enums.ai;

import lombok.AllArgsConstructor;
import lombok.Getter;

@Getter
@AllArgsConstructor
public enum ModelTypeEnum {
    CHAT(0, "对话"),
    CHAT_IMAGE(1, "对话、图像理解"),
    CHAT_IMAGE_FILE(2, "对话、图像理解、文件分析"),
    CHAT_MUSIC(3, "音乐生成"),
    CHAT_PPT(4, "PPT生成");
    private final Integer code;
    private final String desc;

    public static ModelTypeEnum getByCode(Integer modelType) {
        for (ModelTypeEnum modelTypeEnum : values()) {
            if (modelTypeEnum.getCode().equals(modelType)) {
                return modelTypeEnum;
            }
        }
        return null;
    }
}
